Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add FP16 support #257

Open
wants to merge 6 commits into
base: master
Choose a base branch
from
Open

Add FP16 support #257

wants to merge 6 commits into from

Conversation

micahcc
Copy link

@micahcc micahcc commented Dec 28, 2024

  • Adds support for floating point-16 bit images
  • Use the half crate since half precision floats are nightly only
  • added several tests of different features, including prediction

src/decoder/mod.rs Outdated Show resolved Hide resolved
Copy link
Contributor

@fintelia fintelia left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me.

Only concern is whether we have permission to distribute the test image. Could you share the source of the image and the license it is covered under?

@micahcc micahcc force-pushed the main branch 2 times, most recently from a8fdc9d to 77a22d6 Compare January 28, 2025 04:53
@micahcc
Copy link
Author

micahcc commented Jan 28, 2025

@fintelia I removed the previous image which I had pulled from the internet and switched to images generated with gimp. I relinquish all rights.

- add predictor for fp16
- add support for reading half (fp16) images
- add a pure white test image (generated with gimp)
- add predictor tests
@micahcc micahcc force-pushed the main branch 2 times, most recently from 166282b to 2d98a70 Compare January 28, 2025 07:05
@micahcc
Copy link
Author

micahcc commented Jan 31, 2025

@fintelia how do I merge this?

@micahcc
Copy link
Author

micahcc commented Jan 31, 2025

It looks like half doesn't build with 1.61, should this be feature flagged then?

@fintelia
Copy link
Contributor

Should be fine to bump the MSRV to 1.70. That'll require changes in Cargo.toml and .github/workflows/rust.yml

Micah Chambers (eos) added 2 commits January 31, 2025 14:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants